tensor([[[0.3600, 0.0000],
         [0.3600, 0.2400]],

        [[0.0000, 0.5200],
         [0.0000, 0.7600]],

        [[0.6400, 0.4800],
         [0.6400, 0.0000]]])